import os
import glob

def find_cuda_path():
    # 检查常见的安装路径
    possible_paths = [
        r'C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v*',
        r'D:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v*',
        r'D:\NVIDIA\CUDA\v*',
        r'D:\CUDA\v*'
    ]
    
    for pattern in possible_paths:
        paths = glob.glob(pattern)
        if paths:
            print("找到的 CUDA 路径:")
            for path in paths:
                print(path)
            return
            
    print("未找到 CUDA 安装路径")

if __name__ == "__main__":
    find_cuda_path() 